Skip to content

SARIX and GBQR take NSSP Data#14

Merged
lshandross merged 30 commits intomainfrom
ls/sarix-gbqr-takes-nssp-data/9
Nov 24, 2025
Merged

SARIX and GBQR take NSSP Data#14
lshandross merged 30 commits intomainfrom
ls/sarix-gbqr-takes-nssp-data/9

Conversation

@lshandross
Copy link
Contributor

Updates the SARIX and GBQR model classes to take NSSP as a data source, as well as the tests/test data to test the new functionality. Likely most easily reviewed by looking at individual commits, and I could also walk someone through it if need be

* Add tests (and test data) for SARIX and GBQR for NSSP data source
* Rename tests and test data to specify the main data source (i.e. NHSN or NSSP)
* Update file paths in tests to be dynamic (not hard-coded)
@nickreich nickreich closed this Oct 27, 2025
@nickreich nickreich reopened this Oct 27, 2025
matthewcornell
matthewcornell previously approved these changes Oct 28, 2025
* Update sarix fourier tests to use separate states and hsas fields, same logic as other tests, config creation helpers when applicable
* Add args to config creation helper methods to be more flexible
matthewcornell
matthewcornell previously approved these changes Nov 19, 2025
Copy link
Member

@matthewcornell matthewcornell left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I didn't do a deep dive into the logic, but it looks good from a quick-ish scan. Exciting!

nickreich
nickreich previously approved these changes Nov 19, 2025
Copy link
Member

@nickreich nickreich left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

My comments are all minor and should not stop a merge for expediency. If we don't address one of my comments, could we file an issue for it?

Better match original code + Avoid error whenNot all requested locations in data
@lshandross lshandross dismissed stale reviews from nickreich and matthewcornell via a63d488 November 24, 2025 19:14
nickreich
nickreich previously approved these changes Nov 24, 2025
Copy link
Member

@nickreich nickreich left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

reviewed synchronously

New functionality added: allow NSSP-based forecasting
Breaking change: run_config.locations superseded by run_config.states and run_config.hsas
Copy link
Member

@nickreich nickreich left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM! thanks for the iteration.

@lshandross lshandross merged commit f0160d4 into main Nov 24, 2025
1 check passed
nickreich added a commit that referenced this pull request Dec 8, 2025
Documents the NSSP data source support added in PR #14, including:
- NSSP data source support for SARIX and GBQR models
- HSA and state-level forecasting
- Breaking change: locations replaced by states/hsas config

Also adds codespell config to ignore "HSA" acronym.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
@lshandross lshandross linked an issue Jan 8, 2026 that may be closed by this pull request
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

generalize SARIXModel to take sources other than NHSN

4 participants